






FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04


# Set environment variables
ENV NVENCODE_CFLAGS "-I/usr/local/cuda/include"
ENV CV_VERSION=4.2.0
ENV DEBIAN_FRONTEND=noninteractive

# Get all dependencies
RUN apt-get update && apt-get install -y \
    git zip unzip libssl-dev libcairo2-dev lsb-release libgoogle-glog-dev libgflags-dev libatlas-base-dev libeigen3-dev software-properties-common \
    build-essential cmake pkg-config libapr1-dev autoconf automake libtool curl libc6 libboost-all-dev debconf libomp5 libstdc++6 \
    libqt5core5a libqt5xml5 libqt5gui5 libqt5widgets5 libqt5concurrent5 libqt5opengl5 libcap2 libusb-1.0-0 libatk-adaptor neovim \
    python3-pip python3-tornado python3-dev python3-numpy python3-virtualenv libpcl-dev libgoogle-glog-dev libgflags-dev libatlas-base-dev \
    libsuitesparse-dev python3-pcl pcl-tools libgtk2.0-dev libavcodec-dev libavformat-dev libswscale-dev libtbb2 libtbb-dev libjpeg-dev \
    libpng-dev libtiff-dev xfce4-terminal &&\
    rm -rf /var/lib/apt/lists/*
#* conda not be installed :
# libdc1394-22-dev

WORKDIR /




RUN pip3 install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118




# RUN  git clone https://gitee.com/xu-shaocong/mtpsl.git

# WORKDIR  /mtpsl

# #* install the repo 

# RUN mkdir logs

# # RUN pip3 install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113



# RUN pip install pytorch-lightning
# RUN pip install omegaconf
# RUN pip install einops
# RUN pip install progress 
# RUN pip install open_clip_torch

# RUN pip install numpy 
# RUN pip install chardet
# RUN pip install matplotlib 
# RUN pip install scipy
# RUN pip install loguru



# #!=================================================================
# #* not working
# # pip install tensorboard==2.6.0
# # pip install tensorboard-data-server==0.7.2
# # pip install tensorboard-plugin-wit==1.6.post3
# # pip install tensorboardX==1.8
# #!=================================================================


# RUN pip install ipython
# RUN pip install jupyter 

# #* for train
# RUN pip install transformers
# RUN pip install open_clip_torch


# RUN pip3 install opencv-python





